#pragma once

#include "Reactor.hpp"
#include "Util.hpp"
#include <cerrno>
#include <string>
#include <vector>

#define ONCE_SIZE 128

// 1: 本轮读取全部完成
//-1: 读取出错
// 0: 对端关闭链接
static int RecverCore(int sock, std::string &inbuffer)
{
    while (true)
    {
        char buffer[ONCE_SIZE];
        ssize_t s = recv(sock, buffer, ONCE_SIZE - 1, 0);
        if (s > 0)
        {
            buffer[s] = '\0';
            // 读取成功
            inbuffer += buffer;
        }
        else if (s < 0)
        {
            // IO被信号打断，概率特别低
            if (errno == EINTR)
            {
                continue;
            }
            // 读完，底层没数据了
            if (errno == EAGAIN || errno == EWOULDBLOCK)
            {
                return 1; // success
            }
            // 真的出错了
            return -1;
        }
        else // s == 0
        {
            return 0;
        }
    }
}

// 1: 全部将数据发送完成
// 0: 数据没有发完，但是不能再发了
//-1: 发送失败
int SenderCore(int sock, std::string &outbuffer)
{
    while (true)
    {
        int total = 0; // 本轮累计发送的数据量
        const char *start = outbuffer.c_str();
        int size = outbuffer.size();
        ssize_t curr = send(sock, start + total, size - total, 0);
        if (curr > 0)
        {
            total += curr;
            if (total == size)
            {
                // 全部将数据发送完成
                outbuffer.clear();
                return 1;
            }
        }
        else
        {
            // 被信号中断了
            if (errno == EINTR)
                continue;
            // 数据没有发完，但是不能再发了！
            if (errno == EAGAIN || errno == EWOULDBLOCK)
            {
                outbuffer.erase(0, total);
                return 0;
            }

            return -1;
        }
    }
}

int Recver(Event *evp)
{
    std::cout << "Recver been called" << std::endl;

    // 1. 真正的读取 --- 将sock下就绪的数据读到sock对应的Event的inbuffer中
    int result = RecverCore(evp->sock, evp->inbuffer);
    if (result <= 0)
    {
        // 差错处理
        if (evp->errorer)
        {
            evp->errorer(evp);
        }
        return -1;
    }
    // 1+2X3+4X5+6X(用X作分隔符)
    // 2. 分包-- 一个或者多个报文 -- 解决粘包问题
    std::vector<std::string> tokens;
    std::string sep = "X";
    SplitSegment(evp->inbuffer, &tokens, sep);
    // 3. 反序列化 -- 针对一个报文 -- 提取有效参与计算或者存储的信息
    for (auto &seg : tokens) // 1+2 3+4
    {
        std::string data1, data2;
        // 就是和业务强相关啦
        if (Deserialize(seg, &data1, &data2))
        {
            // 4. 业务逻辑 -- 得到结果
            int x = atoi(data1.c_str());
            int y = atoi(data2.c_str());
            int z = x + y;
            // 5. 构建响应 -- 之后先添加到evp->outbuffer，再发送
            // 2+3X -> 2+3=5X
            std::string res = data1;
            res += "+";
            res += data2;
            res += "=";
            res += std::to_string(z);
            res += sep;

            evp->outbuffer += res; // 发送数据(放到outbuffer)
        }
    }
    // 6. 进行发送
    //   1 必须条件成熟了(写事件就绪)，你才能发送
    //   2 一般只有将报文处理完毕，才需要发送
    // 所以对于写事件，我们通常是按需设置，即用户就绪才设置文件描述符关心写事件
    // 因为写事件一般基本都是就绪的，但是用户不一定是就绪的，我们如果一开始就把EPOLLOUT设置了，
    // 就会导致事件派发方一直在派发写事件(可是用户没有就绪，没办法写)，对服务器压力太大。所以，
    // 绝大部分情况下，写一个服务器时绝对不能EPOLLIN和EPOLLOUT同时设置，对于写事件要按需设置。
    if (!(evp->outbuffer).empty())
    {
        // 写打开的时候，用户默认就是就绪的！
        // 即便是发送缓冲区已经满了，只要用户重新设者了OUT事件，EPOLLOUT至少会在触发一次！
        evp->R->EnableRW(evp->sock, true, true);
        // Sender负责写关闭
    }
    return 0;
}

int Sender(Event *evp)
{
    std::cout << "Sender been called" << std::endl;

    // 1: 全部将数据发送完成
    // 0: 数据没有发完，但是不能再发了
    //-1: 发送失败
    int result = SenderCore(evp->sock, evp->outbuffer);
    if (result == 1)
    {
        evp->R->EnableRW(evp->sock, true, false); // 按需设置
    }
    else if (result == 0)
    {
        // 你可以什么都不做
        evp->R->EnableRW(evp->sock, true, true); // 按需设置
    }
    else
    {
        if (evp->errorer)
        {
            evp->errorer(evp);
        }
    }
}

int Errorer(Event *evp)
{
    std::cout << "Errorer been called" << std::endl;
    evp->R->DeleteEvent(evp);
}

